from glob import glob
from streamlit_image_viewer import image_viewer
import streamlit as st
def image_viewer_page():
    col0, image_viewer_col, col4 = st.columns([1,5,1])
    with image_viewer_col:
        st.title("数据集预览")
        # st.title("Sample Images")
        image_path_list = glob('./image/*.jpg') + glob('./image/*.png') 
        image_viewer(image_path_list, ncol=5, nrow=2, image_name_visible=False, key="image_viewer")

if __name__ == "__main__":
    image_viewer_page()